# -*- coding: utf-8 -*-
# Time: 2022/6/5 23:30
import os
import sys
import logging
from datetime import datetime
from logging import handlers


# 默认的配置
# LOG_LEVEL = logging.DEBUG   # 默认等级
LOG_LEVEL = logging.INFO   # 默认等级
LOG_FMT = '%(asctime)s %(filename)s [line:%(lineno)d] %(levelname)s: %(message)s'  # 默认输出格式
LOG_DATE_FMT = '%Y-%m-%d %H:%M:%S'  # 默认时间格式
LOG_WHEN = "D"


class Logger(object):

    def __init__(self, log="log"):
        LOG_FILEPATH = f"./logs/{log}.log"
        # 1. 获取一个logger对象
        self._logger = logging.getLogger()
        # 2. 设置format对象
        self.formatter = logging.Formatter(fmt=LOG_FMT, datefmt=LOG_DATE_FMT)
        # 3.设置日志输出
        # 3.1 设置文件日志模式
        self._logger.addHandler(self._get_file_handler(LOG_FILEPATH))
        # 3.2 设置终端日志模式
        # self._logger.addHandler(self._get_console_handler())
        # 4. 设置日志等级
        self._logger.setLevel(LOG_LEVEL)

    def _get_file_handler(self, filename, when=LOG_WHEN, back_count=3):
        """返回一个文件日志handler"""
        f_dir, f_name = os.path.split(filename)
        os.makedirs(f_dir, exist_ok=True)  # 当前目录新建log文件夹

        # 1. 获取一个文件日志handler
        # 往文件里写入#指定间隔时间自动生成文件的处理器
        filehandler = handlers.TimedRotatingFileHandler(filename=filename, when=when, encoding='utf-8')
        # 2. 设置日志格式
        filehandler.setFormatter(self.formatter)
        def namer(filename):
            dir_name, base_name = os.path.split(filename)
            base_name = "{}{}.log".format(base_name.replace('.log.', ""), datetime.now().strftime("%Y-%m-%d_%H%M%S"))
            rotation_filename = os.path.join(dir_name, base_name)
            return rotation_filename

        filehandler.namer = namer
        filehandler.suffix = "_"
        # 3. 返回
        return filehandler

    def _get_console_handler(self):
        """返回一个输出到终端日志handler"""
        # 1. 获取一个输出到终端日志handler
        console_handler = logging.StreamHandler(sys.stdout)
        # 2. 设置日志格式
        console_handler.setFormatter(self.formatter)
        # 3. 返回handler
        return console_handler

    @classmethod
    def logger(cls, log="log"):
        return cls(log)._logger


# 初始化并配一个logger对象,达到单例的
# 使用时,直接导入logger就可以使用
logger = Logger.logger

if __name__ == '__main__':
    logger(__name__)


